package cn.dlc.com.sys.config.datasource;

/**
 * @Author: dlc
 * @Date: 2021/12/4 - 10:01
 */
public class DataSourceHandler {

    /**
     * 弄一个守护线程来切换
     */
    public static final ThreadLocal<String> holder = new ThreadLocal<>();

    /**
     * 项目启动的时候将读写数据源添加到 holder 里面
     * @param dataSourceName
     */
    public static void putDataSource(String dataSourceName){
        holder.set(dataSourceName);
    }

    /**
     * 获取数据源
     * @return
     */
    public static String getDataSource(){
        return holder.get();
    }
}
